package features.advance.leetcode.compilers;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * @author LIN
 * @date 2022-04-18 09:32
 *  819. 最常见的单词
 *  难度：简单
 * 给定一个段落 (paragraph) 和一个禁用单词列表 (banned)。返回出现次数最多，同时不在禁用列表中的单词。
 *
 * 题目保证至少有一个词不在禁用列表中，而且答案唯一。
 *
 * 禁用列表中的单词用小写字母表示，不含标点符号。段落中的单词不区分大小写。答案都是小写字母。
 *
 *
 *
 * 示例：
 *
 * 输入:
 * paragraph = "Bob hit a ball, the hit BALL flew far after it was hit."
 * banned = ["hit"]
 * 输出: "ball"
 * 解释:
 * "hit" 出现了3次，但它是一个禁用的单词。
 * "ball" 出现了2次 (同时没有其他单词出现2次)，所以它是段落里出现次数最多的，且不在禁用列表中的单词。
 * 注意，所有这些单词在段落里不区分大小写，标点符号需要忽略（即使是紧挨着单词也忽略， 比如 "ball,"），
 * "hit"不是最终的答案，虽然它出现次数更多，但它在禁用单词列表中。
 *
 *
 * 提示：
 *
 * 1 <= 段落长度 <= 1000
 * 0 <= 禁用单词个数 <= 100
 * 1 <= 禁用单词长度 <= 10
 * 答案是唯一的, 且都是小写字母 (即使在 paragraph 里是大写的，即使是一些特定的名词，答案都是小写的。)
 * paragraph 只包含字母、空格和下列标点符号!?',;.
 * 不存在没有连字符或者带有连字符的单词。
 * 单词里只包含字母，不会出现省略号或者其他标点符号。
 */
public class Solution819 {

    public static void main(String[] args) {
        Solution solution = new Solution() {
        };
        String paragraph = "Bob hit a ball, the hit BALL flew far after it was hit.";
        String[] banned = {"hit"};
        String mostCommonWord = solution.mostCommonWord(paragraph, banned);
        System.out.println(mostCommonWord);

    }

    static class Solution {
        public String mostCommonWord(String paragraph, String[] banned) {
            Set<String> bannedSet = Arrays.stream(banned).map(e -> e).collect(Collectors.toSet());
            if (paragraph == null || paragraph.length() == 0) {
                return paragraph;
            } else {
                Map<String, Integer> map = new HashMap(16);
                StringBuilder sb = new StringBuilder();
                char[] arr = paragraph.toCharArray();
                String res = null;
                int max = -1;
                for (int i = 0; i < arr.length; i++) {
                    char a = arr[i];
                    if (sb.length() > 0 && (a == ' ' || a == ';' || a == '\'' || a == '?' || a == '!' || a == ',' || a == '.')) {
                        String key = sb.toString().toLowerCase();
                        if (!bannedSet.contains(key)) {
                            int count = map.containsKey(key) ? map.get(key) : 0;

                            map.put(key, ++count);
                            if (max <= count) {
                                max = count;
                                res = key;
                            }
                        }

                        sb = new StringBuilder();
                    } else if (a != ';' && a != '\'' && a != '?' && a != '!' && a != ',' && a != '.' && a != ' ') {
                        sb.append(a);
                    }
                }
                if (sb.length() > 0) {
                    String key = sb.toString().toLowerCase();
                    if (!bannedSet.contains(key)) {
                        int count = map.getOrDefault(key, 0);

                        map.put(key, ++count);
                        if (max <= count) {
                            max = count;
                            res = key;
                        }
                    }
                }
                System.out.println(map);
                return res;
            }
        }
    }
}
